_base_ = ["./voc0712.py", "./schedule_2x.py", "./default_runtime.py"]
# model setting
backbone_pretrained_path = ("https://download.pytorch.org/models"
                            "/shufflenetv2_x1-5666bf0f80.pth")
model = dict(type="CenterNet",
             pretrained=backbone_pretrained_path,
             backbone=dict(
                 type="ShuffleNetv2",
                 ratio=1.0,
                 out_stage_names=["conv5"]),
             neck=dict(type="DeconvUP",
                       in_channels_list=[1024],
                       out_channels=256),
             bbox_head=dict(type="CenterHead",
                            num_classes=20,
                            in_channels=256,
                            loss_heatmap=dict(type="GaussianFocalLoss",
                                              alpha=2.0,
                                              gamma=4.0,
                                              loss_weight=1),
                            loss_wh_size=dict(type="L1Loss",
                                              reduction="mean",
                                              loss_weight=0.1),
                            loss_ct_off=dict(type="L1Loss",
                                             reduction="mean",
                                             loss_weight=1)))
train_cfg = None
test_cfg = None
